//
// Author:
//   Jb Evain (jbevain@gmail.com)
//
// Copyright (c) 2008 - 2015 Jb Evain
// Copyright (c) 2008 - 2011 Novell, Inc.
//
// Licensed under the MIT/X11 license.
//

using System;
using System.Collections.Generic;
using System.Text;

namespace MonoFN.Cecil.Rocks
{
    public class DocCommentId
    {
        private StringBuilder id;

        private DocCommentId()
        {
            id = new();
        }

        private void WriteField(FieldDefinition field)
        {
            WriteDefinition('F', field);
        }

        private void WriteEvent(EventDefinition @event)
        {
            WriteDefinition('E', @event);
        }

        private void WriteType(TypeDefinition type)
        {
            id.Append('T').Append(':');
            WriteTypeFullName(type);
        }

        private void WriteMethod(MethodDefinition method)
        {
            WriteDefinition('M', method);

            if (method.HasGenericParameters)
            {
                id.Append('`').Append('`');
                id.Append(method.GenericParameters.Count);
            }

            if (method.HasParameters)
                WriteParameters(method.Parameters);

            if (IsConversionOperator(method))
                WriteReturnType(method);
        }

        private static bool IsConversionOperator(MethodDefinition self)
        {
            if (self == null)
                throw new ArgumentNullException("self");

            return self.IsSpecialName && (self.Name == "op_Explicit" || self.Name == "op_Implicit");
        }

        private void WriteReturnType(MethodDefinition method)
        {
            id.Append('~');
            WriteTypeSignature(method.ReturnType);
        }

        private void WriteProperty(PropertyDefinition property)
        {
            WriteDefinition('P', property);

            if (property.HasParameters)
                WriteParameters(property.Parameters);
        }

        private void WriteParameters(IList<ParameterDefinition> parameters)
        {
            id.Append('(');
            WriteList(parameters, p => WriteTypeSignature(p.ParameterType));
            id.Append(')');
        }

        private void WriteTypeSignature(TypeReference type)
        {
            switch (type.MetadataType)
            {
                case MetadataType.Array:
                    WriteArrayTypeSignature((ArrayType)type);
                    break;
                case MetadataType.ByReference:
                    WriteTypeSignature(((ByReferenceType)type).ElementType);
                    id.Append('@');
                    break;
                case MetadataType.FunctionPointer:
                    WriteFunctionPointerTypeSignature((FunctionPointerType)type);
                    break;
                case MetadataType.GenericInstance:
                    WriteGenericInstanceTypeSignature((GenericInstanceType)type);
                    break;
                case MetadataType.Var:
                    id.Append('`');
                    id.Append(((GenericParameter)type).Position);
                    break;
                case MetadataType.MVar:
                    id.Append('`').Append('`');
                    id.Append(((GenericParameter)type).Position);
                    break;
                case MetadataType.OptionalModifier:
                    WriteModiferTypeSignature((OptionalModifierType)type, '!');
                    break;
                case MetadataType.RequiredModifier:
                    WriteModiferTypeSignature((RequiredModifierType)type, '|');
                    break;
                case MetadataType.Pointer:
                    WriteTypeSignature(((PointerType)type).ElementType);
                    id.Append('*');
                    break;
                default:
                    WriteTypeFullName(type);
                    break;
            }
        }

        private void WriteGenericInstanceTypeSignature(GenericInstanceType type)
        {
            if (type.ElementType.IsTypeSpecification())
                throw new NotSupportedException();

            WriteTypeFullName(type.ElementType, stripGenericArity: true);
            id.Append('{');
            WriteList(type.GenericArguments, WriteTypeSignature);
            id.Append('}');
        }

        private void WriteList<T>(IList<T> list, Action<T> action)
        {
            for (int i = 0; i < list.Count; i++)
            {
                if (i > 0)
                    id.Append(',');

                action(list[i]);
            }
        }

        private void WriteModiferTypeSignature(IModifierType type, char id)
        {
            WriteTypeSignature(type.ElementType);
            this.id.Append(id);
            WriteTypeSignature(type.ModifierType);
        }

        private void WriteFunctionPointerTypeSignature(FunctionPointerType type)
        {
            id.Append("=FUNC:");
            WriteTypeSignature(type.ReturnType);

            if (type.HasParameters)
                WriteParameters(type.Parameters);
        }

        private void WriteArrayTypeSignature(ArrayType type)
        {
            WriteTypeSignature(type.ElementType);

            if (type.IsVector)
            {
                id.Append("[]");
                return;
            }

            id.Append("[");

            WriteList(type.Dimensions, dimension =>
            {
                if (dimension.LowerBound.HasValue)
                    id.Append(dimension.LowerBound.Value);

                id.Append(':');

                if (dimension.UpperBound.HasValue)
                    id.Append(dimension.UpperBound.Value - (dimension.LowerBound.GetValueOrDefault() + 1));
            });

            id.Append("]");
        }

        private void WriteDefinition(char id, IMemberDefinition member)
        {
            this.id.Append(id).Append(':');

            WriteTypeFullName(member.DeclaringType);
            this.id.Append('.');
            WriteItemName(member.Name);
        }

        private void WriteTypeFullName(TypeReference type, bool stripGenericArity = false)
        {
            if (type.DeclaringType != null)
            {
                WriteTypeFullName(type.DeclaringType);
                id.Append('.');
            }

            if (!string.IsNullOrEmpty(type.Namespace))
            {
                id.Append(type.Namespace);
                id.Append('.');
            }

            var name = type.Name;

            if (stripGenericArity)
            {
                var index = name.LastIndexOf('`');
                if (index > 0)
                    name = name.Substring(0, index);
            }

            id.Append(name);
        }

        private void WriteItemName(string name)
        {
            id.Append(name.Replace('.', '#').Replace('<', '{').Replace('>', '}'));
        }

        public override string ToString()
        {
            return id.ToString();
        }

        public static string GetDocCommentId(IMemberDefinition member)
        {
            if (member == null)
                throw new ArgumentNullException("member");

            var documentId = new DocCommentId();

            switch (member.MetadataToken.TokenType)
            {
                case TokenType.Field:
                    documentId.WriteField((FieldDefinition)member);
                    break;
                case TokenType.Method:
                    documentId.WriteMethod((MethodDefinition)member);
                    break;
                case TokenType.TypeDef:
                    documentId.WriteType((TypeDefinition)member);
                    break;
                case TokenType.Event:
                    documentId.WriteEvent((EventDefinition)member);
                    break;
                case TokenType.Property:
                    documentId.WriteProperty((PropertyDefinition)member);
                    break;
                default:
                    throw new NotSupportedException(member.FullName);
            }

            return documentId.ToString();
        }
    }
}